package club.xiaojiawei.back;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 5/20/22 9:30 PM
 * @question 39. 组合总和
 * @description 给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。
 * candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。
 * 对于给定的输入，保证和为 target 的不同组合数少于 150 个。
 */
public class CombinationSum39 {

    public static void main(String[] args) {
        CombinationSum39 test = new CombinationSum39();
        List<List<Integer>> result = test.combinationSum(new int[]{2,7,6,3,5,1}, 9);
        System.out.println(result);
    }

    /**
     * 排序+递归
     * @param candidates
     * @param target
     * @return
     */
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);
        recursion(candidates, target, 0);
        return result;
    }

    List<List<Integer>> result = new ArrayList<>();

    List<Integer> temp = new ArrayList<>();

    int count = 0;

    public void recursion(int[] candidates, int target, int startIndex){
        for (int i = startIndex; i < candidates.length && count + candidates[i] <= target; i++){
            count += candidates[i];
            temp.add(candidates[i]);
            if (count == target){
                result.add(new ArrayList<>(temp));
            }else {
                recursion(candidates, target, i);
            }
            count -= candidates[i];
            temp.remove(temp.size() - 1);
        }
    }

    /**
     * 官方-搜索回溯
     * @param candidates
     * @param target
     * @return
     */
    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        List<Integer> combine = new ArrayList<Integer>();
        dfs(candidates, target, ans, combine, 0);
        return ans;
    }

    public void dfs(int[] candidates, int target, List<List<Integer>> ans, List<Integer> combine, int idx) {
        if (idx == candidates.length) {
            return;
        }
        if (target == 0) {
            ans.add(new ArrayList<Integer>(combine));
            return;
        }
        // 看下一个元素是否满足
        dfs(candidates, target, ans, combine, idx + 1);
        // 看当前元素是否满足
        if (target - candidates[idx] >= 0) {
            combine.add(candidates[idx]);
            dfs(candidates, target - candidates[idx], ans, combine, idx);
            combine.remove(combine.size() - 1);
        }
    }

}
